import os
from dotenv import load_dotenv

# 加载 .env 文件中的环境变量
load_dotenv()

# 从环境变量中获取 OpenAI API Key
# 请在项目根目录下创建一个名为 .env 的文件
# 并在其中添加一行: OPENAI_API_KEY='your_api_key_here'
OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")

# 定义输出目录
BASE_DIR = os.path.dirname(os.path.abspath(__file__))
OUTPUTS_DIR = os.path.join(BASE_DIR, 'outputs')
VIDEOS_DIR = os.path.join(OUTPUTS_DIR, 'videos')

# 确保所有输出目录都存在
def create_directories():
    os.makedirs(VIDEOS_DIR, exist_ok=True)